export EVAL_ERTACACHE=0
export SAVE_ERROR=0
export CACHE_ADJUST=0
export SAVE_RESIDUAL=0
export TIMESTEP_ALIGNMENT=0
export TEST_SPEED=0
export SKIP_LIST="[2, 4, 5, 6, 7, 9, 10, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 23, 24, 25]"
export DATA_TEST=2
export CALIBRATE_REL_L1_THRESH=0.6
export PROJECT_DOLDER="your_folder/ERTACache"
export PYTHONPATH=$PROJECT_DOLDER:$PYTHONPATH
export PROMPT_PATH="your_data_folder/mscoco_val2014_30k/metadata.csv"
export CKPT_DIR="your_model_folder/black-forest-labs-FLUX.1-dev"
export PYTHONPATH=$PROJECT_DOLDER:$PYTHONPATH

###### original
export TIMESTEP_ALIGNMENT=0
export EVAL_ERTACACHE=0
export SAVE_ERROR=1
export OUTPUT_DIR="./sample/original"
CUDA_VISIBLE_DEVICES=0 python3 ertacache_flux.py


# ###### eval ertacache + timealign
export TIMESTEP_ALIGNMENT=1
export EVAL_ERTACACHE=1
export SAVE_ERROR=1
export OUTPUT_DIR="./sample/ertacache_timealign"
CUDA_VISIBLE_DEVICES=0 python3 ertacache_flux.py


# ###### eval ertacache + timealign + timeadjust
export DATA_FOLDER_ORI="./sample/original/tensor"
export DATA_FOLDER_CACHE="./sample/ertacache_timealign/ertacache_tensor"
python3 $PROJECT_DOLDER/adjust_utils/adjust_data_analysis.py
export TIMESTEP_ALIGNMENT=1
export EVAL_ERTACACHE=1
export SAVE_ERROR=0
export ADJUST_FOLDER="./sample/ertacache_timealign"
export CACHE_ADJUST=1
export OUTPUT_DIR="./sample/ertacache_timealign_adjust"
CUDA_VISIBLE_DEVICES=0 python3 ertacache_flux.py


